#!/usr/bin env python
# -*- coding: utf-8 -*-
###########################################################################
#   Copyright (C) 2008 by Serj Poltavsky                                  #
#   serge.uliss at gmail.com                                              #
#                                                                         #
#   This program is free software; you can redistribute it and/or modify  #
#   it under the terms of the GNU General Public License as published by  #
#   the Free Software Foundation; either version 3 of the License, or     #
#   (at your option) any later version.                                   #
#                                                                         #
#   This program is distributed in the hope that it will be useful,       #
#   but WITHOUT ANY WARRANTY; without even the implied warranty of        #
#   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         #
#   GNU General Public License for more details.                          #
#                                                                         #
#   You should have received a copy of the GNU General Public License     #
#   along with this program. If not, see <http://www.gnu.org/licenses/>   #
###########################################################################

import os
import sys
from dvdobject import DvdContainer
from dvdmenu import DvdMenu
from dvdauthor import *

class Dvd(DvdContainer):
    def __init__(self, format):
        DvdContainer.__init__(self, format)
        if not os.path.exists(self.tmp_dir()):
            os.mkdir(self.tmp_dir())

        self.menu = DvdMenu(format)

    def get_menu(self):
        return self.__menu

    def export(self):
        self.dvdauthor = DvdAuthor()
        self.export_menu()
        self.export_titles()
        self.dvdauthor.update()
        self.dvdauthor.save(self.tmp_name())

        if __debug__:
            print >>sys.stderr, self.dvdauthor

    def export_menu(self):
        if self.menu:
            self.dvdauthor.vmgm.menu = Menu()
            #self.dvdauthor.vmgm.menu.video = Video()
            #self.dvdauthor.vmgm.menu.video.format = self.format.format
            self.menu.export(self.dvdauthor.vmgm.menu)

    def export_titles(self):
        for title in self:
            t = TitleSet()
            title.export(t)
            self.dvdauthor.append(t)

    def set_menu(self, dvdmenu):
        if dvdmenu is None or isinstance(dvdmenu, DvdMenu):
            self.__menu = dvdmenu
        else:
            raise TypeError("DvdMenu instance expected")

    def tmp_name(self):
        return os.path.join(self.tmp_dir(), "dvdauthor.xml")

    menu = property(get_menu, set_menu)